//===-- ProcessMachCore.h ---------------------------------------*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#ifndef liblldb_ProcessMachCore_h_
#define liblldb_ProcessMachCore_h_

// C Includes
// C++ Includes
#include <list>
#include <vector>

// Other libraries and framework includes
// Project includes
#include "lldb/Core/ConstString.h"
#include "lldb/Core/Error.h"
#include "lldb/Target/Process.h"

class ThreadKDP;

class ProcessMachCore : public lldb_private::Process
{
public:
    //------------------------------------------------------------------
    // Constructors and Destructors
    //------------------------------------------------------------------
    ProcessMachCore(lldb::TargetSP target_sp, 
                    lldb::ListenerSP listener,
                    const lldb_private::FileSpec &core_file);
    
    ~ProcessMachCore() override;
    
    static lldb::ProcessSP
    CreateInstance (lldb::TargetSP target_sp, 
                    lldb::ListenerSP listener,
                    const lldb_private::FileSpec *crash_file_path);
    
    static void
    Initialize();
    
    static void
    Terminate();
    
    static lldb_private::ConstString
    GetPluginNameStatic();
    
    static const char *
    GetPluginDescriptionStatic();
    
    //------------------------------------------------------------------
    // Check if a given Process
    //------------------------------------------------------------------
    bool
    CanDebug (lldb::TargetSP target_sp,
              bool plugin_specified_by_name) override;
    
    //------------------------------------------------------------------
    // Creating a new process, or attaching to an existing one
    //------------------------------------------------------------------
    lldb_private::Error
    DoLoadCore () override;
    
    lldb_private::DynamicLoader *
    GetDynamicLoader () override;

    //------------------------------------------------------------------
    // PluginInterface protocol
    //------------------------------------------------------------------
    lldb_private::ConstString
    GetPluginName() override;
    
    uint32_t
    GetPluginVersion() override;
    
    //------------------------------------------------------------------
    // Process Control
    //------------------------------------------------------------------    
    lldb_private::Error
    DoDestroy () override;
    
    void
    RefreshStateAfterStop() override;
    
    //------------------------------------------------------------------
    // Process Queries
    //------------------------------------------------------------------
    bool
    IsAlive () override;

    bool
    WarnBeforeDetach () const override;

    //------------------------------------------------------------------
    // Process Memory
    //------------------------------------------------------------------
    size_t
    ReadMemory (lldb::addr_t addr, void *buf, size_t size, lldb_private::Error &error) override;
    
    size_t
    DoReadMemory (lldb::addr_t addr, void *buf, size_t size, lldb_private::Error &error) override;
    
    lldb::addr_t
    GetImageInfoAddress () override;

protected:
    friend class ThreadMachCore;
    
    void
    Clear ( );
    
    bool
    UpdateThreadList (lldb_private::ThreadList &old_thread_list, 
                      lldb_private::ThreadList &new_thread_list) override;
    
    lldb_private::ObjectFile *
    GetCoreObjectFile ();
private:
    bool 
    GetDynamicLoaderAddress (lldb::addr_t addr);

    typedef enum CorefilePreference { eUserProcessCorefile, eKernelCorefile } CorefilePreferences;

    //------------------------------------------------------------------
    /// If a core file can be interpreted multiple ways, this establishes
    /// which style wins.
    ///
    /// If a core file contains both a kernel binary and a user-process
    /// dynamic loader, lldb needs to pick one over the other.  This could
    /// be a kernel corefile that happens to have a coyp of dyld in its
    /// memory.  Or it could be a user process coredump of lldb while doing
    /// kernel debugging - so a copy of the kernel is in its heap.  This
    /// should become a setting so it can be over-ridden when necessary.
    //------------------------------------------------------------------
    CorefilePreference
    GetCorefilePreference ()
    {
        // For now, if both user process and kernel binaries a present,
        // assume this is a kernel coredump which has a copy of a user
        // process dyld in one of its pages.
        return eKernelCorefile;
    }

    //------------------------------------------------------------------
    // For ProcessMachCore only
    //------------------------------------------------------------------
    typedef lldb_private::Range<lldb::addr_t, lldb::addr_t> FileRange;
    typedef lldb_private::RangeDataVector<lldb::addr_t, lldb::addr_t, FileRange> VMRangeToFileOffset;

    VMRangeToFileOffset m_core_aranges;
    lldb::ModuleSP m_core_module_sp;
    lldb_private::FileSpec m_core_file;
    lldb::addr_t m_dyld_addr;
    lldb::addr_t m_mach_kernel_addr;
    lldb_private::ConstString m_dyld_plugin_name;

    DISALLOW_COPY_AND_ASSIGN (ProcessMachCore);
};

#endif // liblldb_ProcessMachCore_h_
